make a function in python